Skip to content

[LTX2] Add LTX2 Transformer#341

Open
prishajain1 wants to merge 1 commit intomainfrom
prisha/ltx2_transformer
Open

[LTX2] Add LTX2 Transformer#341
prishajain1 wants to merge 1 commit intomainfrom
prisha/ltx2_transformer

Conversation

@prishajain1
Copy link
Collaborator

This PR adds the transformer component for LTX2 Model along with the unittests for the same

New files added:

  • transformer_ltx2.py : Transformer component for LTX2
  • test_transformer_ltx2.py : unittests for transformer
  • ltx2_video.yml : config file for ltx2_video.yml

Files Modified:

  • common_types.py, pyconfig.py : for adding LTX2_VIDEO to the list of allowed model names for inference
  • embeddings_flax.py
    • Refactored sinusoidal timestep embedding generation into a dedicated nnx.Module called NNXTimesteps for better modularity.
    • A new module NNXPixArtAlphaCombinedTimestepSizeEmbeddings combines timestep embeddings with optional resolution and aspect ratio embeddings

@prishajain1 prishajain1 requested a review from entrpn as a code owner March 3, 2026 12:03
@github-actions
Copy link

github-actions bot commented Mar 3, 2026

@prishajain1 prishajain1 marked this pull request as draft March 3, 2026 12:03
@prishajain1 prishajain1 requested a review from mbohlool March 4, 2026 03:39
@prishajain1 prishajain1 force-pushed the prisha/ltx2_transformer branch from 09f17ac to 672a817 Compare March 6, 2026 03:28
@prishajain1 prishajain1 marked this pull request as ready for review March 6, 2026 04:19
@prishajain1 prishajain1 changed the title Add LTX2 Transformer [LTX2] Add LTX2 Transformer Mar 6, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant